Neural network hyper parameter tuning in python